import os
import sys
import pandas as pd
import datetime
from MainConsolidator import MainConsolidator

class handler(object):
    @staticmethod
    def module_name():
        return 'КПК Заготовка колесная'
    @staticmethod
    def handle(row:pd.Series,consolidator:MainConsolidator)->pd.Series:
        weigth_s="-"
        diameter=0
        standart="-"
        try:
            
            diameter = consolidator.scalar_in_guid("Диаметры колес",f"`Производство`=='КПК Колесная заготовка' and `Диаметры SAP`=={row['Диаметр, мм <Производство>']}","Диаметры УП") 
            weigth = consolidator.nearest_in_guid("Параметры по Продукту КПК",
                                                  f"`Производство`=='КПК Колесная заготовка' and `Доп параметр1`=={diameter}",
                                                  "Доп параметр2",
                                                  row["Теоретический вес колесной заготовки, кг <Производство>"],
                                                  "Доп параметр2")
            weigth_s=str(weigth).replace(".",",").replace(",0","")
            standart = consolidator.scalar_in_guid("Стандарты колесной заготовки",f"`Вес`=={weigth}","Стандарт для колесной заготовки")
            print(f"{row['Диаметр, мм <Производство>']} -> {diameter}: {row['Теоретический вес колесной заготовки, кг <Производство>']} -> {weigth_s} : {standart}")
            if pd.isna(diameter) or pd.isna(standart): raise Exception("Не определен вес или стандарт")
            row["error"] = pd.NA
        except Exception as exp: 
            row["error"] = f"{exp}"
        finally:
            row["Продукт УП"]=f"КОЛЕС ЗАГОТОВКА Д{diameter} В{weigth_s} {standart}"
            if  pd.notna(diameter) and diameter==473: row["Продукт УП"]="ИНСТРУМЕНТ ДЕФОРМАЦИИ"
            row["Время укрупнения"]=datetime.datetime.now()
            return row
        
    def debug_handle(self,inputDf:pd.DataFrame,consolidator:MainConsolidator,module:str=None)->pd.DataFrame:
        try:
            if module is None: return inputDf.apply(lambda row: self.handle(row,consolidator), axis=1)
            else: return inputDf[inputDf["Производство2"]==module].apply(lambda row: self.handle(row,consolidator), axis=1)
        except: raise
    
    def get_compiled(self):
        code:str=f"print('{__file__}')"
        try:
            with open(__file__,encoding="utf-8",mode="r") as f:
                code=f.read()
        except Exception as exp: 
            code=f"print('error: {exp}')"
        finally:
            cc = compile(code,"","exec")
            return cc

if __name__=="__billet_wheel__":
    consolidator:MainConsolidator
    record:pd.Series
    result:pd.Series
    result = handler.handle(record,consolidator)

